package ModuleBuilder.UI.tree;

import java.util.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.*;
import javax.swing.event.*;
import javax.swing.border.EtchedBorder;

import java.awt.*;
import java.awt.event.*;
import java.io.*;
import ModuleBuilder.*;
import ModuleBuilder.model.*;	
import ModuleBuilder.UI.*;
import java.beans.*;

/**
 * This is a container class that holds the unit name and the unit ImageIcon
 * In the future this will hold the view for a whole UNIT (multi-stand or
 * single stand)
 */
public class BuilderUnitTrayView {
  String name;
  ImageElement imageElement;

  public BuilderUnitTrayView(String n, ImageElement el) {
    name = n;
	imageElement = el;
  }

  public ImageIcon getUnitIcon() {
    return ( imageElement.getSmallIcon() );
  }

  public String getUnitName() {
    return ( name );
  }

  public ImageElement getElement() {
    return ( imageElement );
  }

  /*
  public Unit getUnit() {
    return();
  }
  */

}